package com.rymcu.forest.config;

import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.Configuration;

/**
 * RAG功能配置类
 */
@Configuration
@ConfigurationProperties(prefix = "rag")
public class RAGConfig {

    /**
     * 是否启用RAG功能
     */
    private boolean enabled = true;

    /**
     * 最多返回相关文章数量
     */
    private int maxArticles = 5;

    /**
     * 相似度阈值
     */
    private double similarityThreshold = 0.7;

    /**
     * 是否启用SuperSQL查询
     */
    private boolean enableSqlQuery = true;

    // Getters and Setters
    public boolean isEnabled() {
        return enabled;
    }

    public void setEnabled(boolean enabled) {
        this.enabled = enabled;
    }

    public int getMaxArticles() {
        return maxArticles;
    }

    public void setMaxArticles(int maxArticles) {
        this.maxArticles = maxArticles;
    }

    public double getSimilarityThreshold() {
        return similarityThreshold;
    }

    public void setSimilarityThreshold(double similarityThreshold) {
        this.similarityThreshold = similarityThreshold;
    }

    public boolean isEnableSqlQuery() {
        return enableSqlQuery;
    }

    public void setEnableSqlQuery(boolean enableSqlQuery) {
        this.enableSqlQuery = enableSqlQuery;
    }
}